'''
统计指定目录大小
'''

# 1、获取指定目录源文件
# 2、读取源文件，获得源文件列表
# 3、源文件列表中的文件若是文件，则获取其大小，并将其文件大小统计计算到一个目录大小变量中
# 4、若该文件是属于文件夹，则需要进行本函数递归操作
# 5、结果返回总文件大小

import os

def get_dir_size(dir):
    # 读取源文件，获得源文件列表
    dirlist = os.listdir(dir)
    # 将总文件大小定义为全局变量
    global sum_file_size
    for f in dirlist:
        filepath = os.path.join(dir,f)
        # 源文件列表下文件若属于文件，则获取其大小
        if os.path.isfile(filepath):
            file_size = os.path.getsize(filepath)
            sum_file_size = sum_file_size + file_size
        # 若列表下该文件是属于文件夹，则需要进行本函数递归操作
        if os.path.isdir(filepath):
            get_dir_size(filepath)

    return sum_file_size

sum_file_size = 0
fileSize = get_dir_size('./MyTest')
print('指定文件夹目录大小为：',fileSize)

